import os
import pandas as pd


def check_and_process_csv_files(folder_path):
    # 遍历指定文件夹中的所有文件
    for filename in os.listdir(folder_path):
        # 检查文件是否为CSV文件
        if filename.endswith('.csv'):
            file_path = os.path.join(folder_path, filename)
            try:
                # 读取CSV文件
                df = pd.read_csv(file_path, encoding='GBK')
                modified_flag = False
                # 检查"井次名称"这一列是否存在
                if '井次名称' in df.columns:
                    print(f"文件 {filename} 包含 '井次名称' 列，正在删除该列。")
                    # 删除"井次名称"这一列
                    df.drop(columns=['井次名称'], inplace=True)
                    modified_flag = True
                else:
                    pass
                if '井名缩写' in df.columns:
                    print(f"文件 {filename} 包含 '井次名称' 列，正在删除该列。")
                    # 删除"井次名称"这一列
                    df.drop(columns=['井名缩写'], inplace=True)
                    modified_flag = True
                # 将修改后的数据保存回文件
                if modified_flag != False:
                    df.to_csv(file_path, index=False, encoding='GBK')
                    print(f"文件 {filename} 已处理并保存。")
            except Exception as e:
                print(f"处理文件 {filename} 时出错: {e}")

            # 使用示例


folder_path = 'section_train_2nd'  # 替换为你的文件夹路径
check_and_process_csv_files(folder_path)

folder_path = 'section_test_2nd'  # 替换为你的文件夹路径
check_and_process_csv_files(folder_path)